import re
import subprocess
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import plotly.graph_objs as go
import gpxpy
import random
from dateutil.parser import parse
from copy import deepcopy
from imageio import imread
from matplotlib.pyplot import imshow
from plotly.offline import init_notebook_mode, iplot
init_notebook_mode(connected=True)
%matplotlib inline
img = imread('basemap.png') / 255
plt.figure(figsize = (10,10))
# Find and show unique colors
img_array = img[:, :, :3].reshape((img.shape[0] * img.shape[1], 3))
colors = np.unique(img_array, axis=0)
n_colors = colors.shape[0]
def show_colors(colors):
# colors_matrix = np.reshape(colors, [4, n_colors // 4, 3])
imshow(np.reshape(colors, (1, -1, 3)), aspect='auto')
plt.xticks([])
plt.yticks([])
plt.gcf().set_size_inches(10, 1)
# Create a custom colormap
color_to_value = {tuple(color[:3]): i / (n_colors - 1) for i, color in enumerate(colors)}
my_cmap_ply = [(value, 'rgb({}, {}, {})'.format(*color)) for color, value in color_to_value.items()]
# Map pixels to values
fun_find_value = lambda x: color_to_value[tuple(x[:3])]
values = np.apply_along_axis(fun_find_value, 2, np.flipud(img))
file = 'Lunch_Run.gpx'
gpx_file = open(file, 'r')
gpx = gpxpy.parse(gpx_file)
df = pd.DataFrame(columns=['lon', 'lat', 'alt', 'time'])
d1 = parse('2020-05-29T11:12:49Z')
for segment in gpx.tracks[0].segments: # all segments
data = segment.points
for point in data:
timesec = (abs((point.time - d1).seconds))
df = df.append({'lon': point.longitude, 'lat' : point.latitude, 'alt' : point.elevation, 'time' : timesec}, ignore_index=True)
traces = []
df.index = df['time']
trace = go.Scatter3d(
x=df['lon'].values, y=df['lat'].values, z=df['time'].values,
showlegend=False
)
traces.append(trace)
yy = np.linspace(52.223, 52.242, img.shape[0])
xx = np.linspace(0.242, 0.284, img.shape[0])
zz = np.full(img.shape[:2], -90)
surf = go.Surface(
x=xx, y=yy, z=zz,
colorscale=my_cmap_ply,
surfacecolor=values,
showscale=False
)
layout = go.Layout(
margin=dict(l=0,r=0,b=0,t=0),
scene=go.layout.Scene(
xaxis=go.layout.scene.XAxis(title='', showticklabels=False),
yaxis=go.layout.scene.YAxis(title='', showticklabels=False),
zaxis=go.layout.scene.ZAxis(title='Time (s)'),
aspectratio=dict(x=1, y=1, z=1.3),
camera=go.layout.scene.Camera(
projection=go.layout.scene.camera.Projection(
type='orthographic'
)
)
)
)
fig = go.Figure(data=[surf] + traces, layout=layout)
iplot(fig, filename=f'adamgoesforarun.html')